package com.liujun.spring4.demo.aop.application.property.crypto;

import sun.misc.BASE64Decoder;
import sun.misc.BASE64Encoder;

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import java.security.Key;
import java.security.SecureRandom;

public class DESUtils {
  private static Key key;
  private static String KEY_STR = "feifei123";

  static {
    try {
      KeyGenerator generator = KeyGenerator.getInstance("DES");
      generator.init(new SecureRandom(KEY_STR.getBytes()));
      key = generator.generateKey();
      generator = null;
    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  /**
   * 对str进行DES加密
   *
   * @param str
   * @return
   */
  public static String getEncryptString(String str) {
    BASE64Encoder base64en = new BASE64Encoder();
    try {
      byte[] strBytes = str.getBytes("UTF8");
      Cipher cipher = Cipher.getInstance("DES");
      cipher.init(Cipher.ENCRYPT_MODE, key);
      byte[] encryptStrBytes = cipher.doFinal(strBytes);
      return base64en.encode(encryptStrBytes);
    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  /**
   * 对str进行DES解密
   *
   * @param str
   * @return
   */
  public static String getDecryptString(String str) {
    BASE64Decoder base64De = new BASE64Decoder();
    try {
      byte[] strBytes = base64De.decodeBuffer(str);
      Cipher cipher = Cipher.getInstance("DES");
      cipher.init(Cipher.DECRYPT_MODE, key);
      byte[] decryptStrBytes = cipher.doFinal(strBytes);
      return new String(decryptStrBytes, "UTF8");
    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  public static void main(String[] args) throws Exception {

    if (null == args || args.length == 0) {
      args = new String[] {"root", "123456"};
    }

    if (args == null || args.length < 1) {
      System.out.println("请输入要加密的字符，用空格分隔.");
    } else {
      for (String arg : args) {
        System.out.println(arg + ":" + getEncryptString(arg));
      }
    }

    // System.out.println(getDecryptString("WnplV/ietfQ="));
    // System.out.println(getDecryptString("gJQ9O+q34qk="));
  }
}
